#ifndef __M_UTIL_H__
#define __M_UTIL_H__
#include "logger.hpp"
#include <iostream>
#include <sstream>
#include <fstream>
#include <string>
#include <vector>
#include <memory>
#include <mysql/mysql.h>
#include <jsoncpp/json/json.h>
#include <websocketpp/server.hpp>
#include <websocketpp/config/asio_no_tls.hpp>

typedef  websocketpp::server<websocketpp::config::asio> wsserver_t;


class mysql_util
{
public:
    static MYSQL *mysql_create(const std::string &host,
                                const std::string &username,
                                const std::string &password,
                                const std::string &dbname,
                                uint16_t port = 3306) 
    {
        //1. 初始化句柄
        MYSQL *mysql = mysql_init(NULL);
        if(mysql == NULL)
        {
            ELOG("mysql init failed!");
            return NULL;
        }
        //2. 链接服务器
        if(mysql_real_connect(mysql,host.c_str(),username.c_str(),password.c_str(),dbname.c_str(),port,NULL,0) == NULL)
        {
            ELOG("connect mysql server failed :%s",mysql_error(mysql));
            return NULL;
        }
        //3. 设置客户端字符集
        if(mysql_set_character_set(mysql,"utf8") != 0)
        {
            ELOG("set client character failed :%s",mysql_error(mysql));
            return NULL;
        }
        return mysql;

    }

    static bool mysql_exec(MYSQL *mysql,const std::string &sql)
    {
        int ret = mysql_query(mysql,sql.c_str());
        if(ret != 0)
        {
            ELOG("%s",sql.c_str());
            ELOG("mysql query failed :%s",mysql_error(mysql));
            return false;
        }
        return true;
    }

    static void mysql_destroy(MYSQL *mysql)
    {
        if(mysql != NULL)
        {
            mysql_close(mysql);
        }
        return ;
    }
};

class json_util
{
public:
    static bool serialize(const Json::Value &root,std::string &str)
    {
        Json::StreamWriterBuilder swb;
        std::unique_ptr<Json::StreamWriter> sw(swb.newStreamWriter());
        std::stringstream ss;
        int ret = sw->write(root,&ss);
        if(ret != 0)
        {
            ELOG("json serialize failed!!");
            return false;
        }
        str = ss.str();
        return true;
    }

    static bool unserialize(const std::string &str,Json::Value &root)
    {
        Json::CharReaderBuilder crb;
        std::unique_ptr<Json::CharReader> cr(crb.newCharReader());
        std::string err;
        bool ret = cr->parse(str.c_str(),str.c_str()+str.size(),&root,&err);
        if(ret == false)
        {
            ELOG("json unserialize failed :%S",err.c_str());
            return false;
        }
        return true;
    }
};

class string_util
{
public:
    static int split(const std::string &src, const std::string &sep, std::vector<std::string> &res)
    {
        size_t pos, idx =0;
        while(idx<src.size())
        {
            pos = src.find(sep,idx);
            if(pos == std::string::npos)
            {
                res.push_back(src.substr(idx));
                break;
            }
            if(pos == idx)
            {
                idx += sep.size();
                continue;
            }
            res.push_back(src.substr(idx,pos-idx));
            idx = pos + sep.size();
        }
        return res.size();
    }
};

class file_util
{
public:
    static bool read(const std::string &filename, std::string &body)
    {
        //1.打开文件
        std::ifstream ifs(filename,std::ios::binary);
        if(ifs.is_open() == false)
        {
            ELOG("%s file open failed!!",filename.c_str());
            return false;
        }
        //2.获取文件大小
        size_t fsize=0;
        ifs.seekg(0,std::ios::end);
        fsize = ifs.tellg();
        ifs.seekg(0,std::ios::beg);
        body.resize(fsize);
        //3.将文件所有数据读取出来
        ifs.read(&body[0],fsize);
        if(ifs.good() == false)
        {
            ELOG("read %s file content failed!!",filename.c_str());
            ifs.close();
            return false;
        }
        //4.关闭文件
        return true;
    }
};

#endif